../../../../models/demos/vanilla_unet/tests/pcc/test_ttnn_unet.py